from PIL import Image
import os

# 定义图片输入和输出文件夹路径
print("hello,我是zhaoh"+"\n"+"为了方便大家批量进行图片的旋转，故编写此程序......"+"\n"+"本人微信:zhaoh904。如果遇到bug 欢迎联系我"+"\n"+"祝生活愉快!")
for i in range(3):
    print("=======================帅气的分隔符=======================")
pic1 = input("从哪里获取图片:")
angle = eval(input("输入旋转的角度"))
pic2 = input("保存在哪里:")
input_folder = pic1
output_folder = pic2

# 若输出文件夹不存在则创建
if not os.path.exists(output_folder):
    os.mkdir(output_folder)

# 遍历输入文件夹中的所有文件
for filename in os.listdir(input_folder):
    file_path = os.path.join(input_folder, filename)
    # 检查文件是否为图片
    if os.path.isfile(file_path) and file_path.lower().endswith(('.png', '.jpg', '.jpeg')):
        try:
            # 打开图片
            with Image.open(file_path) as img:
                # 旋转图片 ？ 度
                rotated_img = img.rotate(angle, expand=True)
                # 构建输出文件路径
                output_path = os.path.join(output_folder, filename)
                # 保存旋转后的图片
                rotated_img.save(output_path)
                print(f"已旋转并保存: {filename}")
        except Exception as e:
            print(f"处理 {filename} 时出错: {e}")